#ifndef AGXKIT_H
#define AGXKIT_H

#include <iostream>
#include <fstream>
#include <string>
using namespace std;

class AGXKit {
private:
    string agxModel;      // AGX 套件型号（固定：AGX Xavier）
    int agxTOPS;          // AI TOPS（固定：32）
    int cudaCores;        // CUDA核心（固定：512）
    int tensorCores;      // Tensor Core（固定：64）
    int memory;           // 显存（固定：32G）
    int storage;          // 存储（固定：32G）

public:
    AGXKit();
    void print() const;
    void save(ofstream& outFile) const;
};

#endif
